#!/bin/bash

GPU_NUM=$1
CFG=$2
DATASETS=$3
OUTPUT_DIR=$4
export TOKENIZERS_PARALLELISM=false
# 指定要使用的 GPU
CUDA_VISIBLE_DEVICES=${GPU_NUM} python -X faulthandler main.py \
        --output_dir ${OUTPUT_DIR} \
        -c ${CFG} \
        --datasets ${DATASETS}  \
        --pretrain_model_path weights/groundingdino_swint_ogc.pth \
        --options text_encoder_type=weights/bert-base-uncased \
        